Skip to content

Add support for JAX Arrays#1

Open
anshradh wants to merge 5 commits intoKiv:mainfrom
anshradh:main
Open

Add support for JAX Arrays#1
anshradh wants to merge 5 commits intoKiv:mainfrom
anshradh:main

Conversation

@anshradh
Copy link

Simple enough change that I figured this was reasonable to add - no worries if you'd prefer to leave this supporting just numpy and torch though!

@anshradh anshradh changed the title (WIP) Add support for JAX Arrays Add support for JAX Arrays Nov 20, 2022
@anshradh anshradh marked this pull request as ready for review November 20, 2022 16:13
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant